#pragma once
#include "compiler.hpp"
#include "runner.hpp"
#include <jsoncpp/json/json.h>

namespace ns_compile_run
{
    using namespace ns_compiler;
    using namespace ns_run;
    class Compile_Run
    {
    public: 
        Compile_Run(){}
        ~Compile_Run(){}
        
        static void RemoveFile(const std::string& file_name)
        {
            std::string _src = PathUtil::Src(file_name);
            if(FileUtil::IsFileExist(_src)) unlink(_src.c_str());

            std::string _exe = PathUtil::Exe(file_name);
            if(FileUtil::IsFileExist(_exe)) unlink(_exe.c_str());

            std::string _comerr = PathUtil::Compile_Error(file_name);
            if(FileUtil::IsFileExist(_comerr)) unlink(_comerr.c_str());

            std::string _stdin = PathUtil::Stdin(file_name);
            if(FileUtil::IsFileExist(_stdin)) unlink(_stdin.c_str());

            std::string _stdout = PathUtil::Stdout(file_name);
            if(FileUtil::IsFileExist(_stdout)) unlink(_stdout.c_str());

            std::string _stderr = PathUtil::Stderr(file_name);
            if(FileUtil::IsFileExist(_stderr)) unlink(_stderr.c_str());
        }

        static std::string CodeToDesc(int status, const std::string& file_name)
        {
            std::string str;
            switch(status)
            {
                case 0:
                str = "运行成功";
                break;
                case 6:
                str = "超出内存范围";
                break;
                case 24:
                str = "超出时间限制";
                break;
                default:
                str = "其他未知错误";
                break;
            }
            return str;
        }

        static void Start(const std::string& in_json, std::string* out_json)
        {
            // 反序列化
            Json::Value in_value;
            Json::Reader reader;
            reader.parse(in_json, in_value);
            std::string code = in_value["code"].asString();
            std::string input = in_value["input"].asString();
            int cpu_limit = in_value["cpu_limit"].asInt();
            int mem_limit = in_value["mem_limit"].asInt();

            std::string file_name;

            // 序列化状态码和错误原因
            Json::Value out_value;
            int status_code = 0;

            if(code.size() == 0)
            {   
                out_value["status"] = -1;
                out_value["reason"] = "代码为空";
                Json::StyledWriter writer;
                *out_json = writer.write(out_value);
                return;
            }

            // 生成唯一文件
            file_name = FileUtil::UniqueFile();
            
            // 写入文件
            if(!FileUtil::WriteFile(PathUtil::Src(file_name), code))
            {
                out_value["status"] = -2;
                out_value["reason"] = "写入文件错误";
                Json::StyledWriter writer;
                *out_json = writer.write(out_value);
                return;
            }

            // 编译
            if(!Compiler::IsCompiler(file_name))
            {
                out_value["status"] = -3;
                out_value["reason"] = "编译文件时代码有错误";
                Json::StyledWriter writer;
                *out_json = writer.write(out_value);
                return;
            }

            // 运行
            int ret = Run::Runner(file_name, cpu_limit, mem_limit);
            if(ret < 0)
            {
                status_code = -2;
            }
            else if(ret > 0)
            {
                status_code = ret;
            }
            else 
            {
                status_code = 0;
                std::string str;
                FileUtil::ReadFile(PathUtil::Stdout(file_name), &str);
                out_value["stdout"] = str;
                FileUtil::ReadFile(PathUtil::Stderr(file_name), &str);
                out_value["stderr"] = str;
            }

            out_value["status"] = status_code;
            out_value["reason"] = CodeToDesc(status_code, file_name);//=0 >0

            Json::StyledWriter writer;
            *out_json = writer.write(out_value);
  
            // 清理文件
            RemoveFile(file_name);
        }
    };
}